import torch.nn as nn

class Autoencoder(nn.Module): 
    def __init__(self): 
        super(Autoencoder, self).__init__() 

        self.encoder = nn.Sequential( 
            nn.Linear(700, 256), 
            nn.ReLU(), 
            nn.Linear(256, 100) ) 

        self.decoder = nn.Sequential( 
            nn.Linear(100, 256), nn.ReLU(), 
            nn.Linear(256, 700), 
            nn.Sigmoid() ) 

    def forward(self, x): 
        encoded = self.encoder(x) 
        decoded = self.decoder(encoded) 
        return decoded
